import xgboost as xgb

from sklearn import tree, ensemble, linear_model
from sklearn import model_selection


CLFS = {
    'lr': linear_model.LogisticRegression(),
    'dt_gini': tree.DecisionTreeClassifier(criterion='gini'),
    'dt_entropy': tree.DecisionTreeClassifier(criterion='entropy'),
    'rf': ensemble.RandomForestClassifier(n_jobs=-1),
    'xgb': xgb.XGBClassifier(n_jobs=-1, max_depth=7, n_estimators=200)
}
